use "/homeKRTK/health_shocks/data_for_main_regs_event_study_20231203.dta", clear
drop if control==1

append using "/homeKRTK/health_shocks/additional_data_for_IV_and_placebo_regs_20240416.dta"
replace treated=1 if d==2
drop if d==1 & control==0

append using "/homeKRTK/health_shocks/additional_data_for_all_regs_controls_d_distrib_as_treated_20250319.dta"

gen NONEMP = 1 if ABS==1
	recode NONEMP (.=0) if ABS==0
	
set seed 12345

keep if d<=6
*keep if d>=3 & d<=6
*keep if NONEMP !=.

gen month_shock = t-12*int(t/12)
	recode month_shock (0=12)
gen month_return = month_shock+d

*keep if treated == 1 | control == 1

forval i=1/8 {
	local j=`i'+8
	rename mean_fq`i'_w w_q`j'
	replace mean_fq`i'_FEj_imp =. if flag_missingFEj_fq`i'==1
	rename mean_fq`i'_FEj_imp FE_q`j'


}
forval i=1/8 {
	local j=8-`i'
	rename mean_lq`i'_w w_q`j'
	replace mean_lq`i'_FEj_imp =. if flag_missingFEj_lq`i'==1
	rename mean_lq`i'_FEj_imp FE_q`j'
}


* reshape *
capture drop FEj_t4 FEj_t5 FEj_t6 FEj_t7 FEj_t8 FEj_t9 FEj_t10 FEj_t11 FEj_t12 
capture drop lnw_restr_t4 lnw_restr_t5 lnw_restr_t6 lnw_restr_t7 lnw_restr_t8 lnw_restr_t9 lnw_restr_t10 lnw_restr_t11 lnw_restr_t12 

reshape long w_q FE_q, i(anon) j(relative_time)


recode teaor1_h1 (-1=99) (.=100)

cap drop date
gen date=t+(relative_time-8)*3
replace date=t+d+(relative_time-9)*3 if relative_time>=8

rename kor kor_orig

rename ev ev_orig
rename t t_orig
rename date t

gen ev=floor((t-1)/12)+2003
merge 1:1 anon t using "/homeProspSSD/Admin3/admin3_alap.dta", nogen keep(master match) keepusing(vallazon1 wtip1 wh1 feor1_h2 kor)
replace wh1=. if wh1>40
rename vallazon1 vallazon
merge m:1 vallazon ev using "/homeKRTK/health_shocks/firmquality.dta", nogen keep(master match)

merge 1:1 anon t using "/homeKRTK/nok_munka/Mental_health/admin3_passziv_ho.dta", nogen keepusing(passziv_ho)
replace passziv_ho=0 if passziv_ho==.

by anon, sort: egen full_d=min(d)
drop if full_d==. //drop anons who are never in our baseline sample

rename t date
rename t_orig t

by anon, sort: egen full_t=min(t)

gen sl_relevant1_=passziv_ho if date>=full_t & date<=full_t+1
by anon, sort: egen sl_relevant1=total(sl_relevant1_)
drop sl_relevant1_

drop if d==. //back to baseline+falsification sample
drop if d>6

keep if sl_relevant1>0 | control==1 //sample restriction!!!

cap drop tfp_wr_w
cap drop w_q_w
cap drop FE_q_w
winsor2 tfp_wr, cuts(1 99) by(ev)
winsor2 w_q, cuts(1 99) by(ev)
winsor2 FE_q, cuts(1 99) by(ev)

gen tr_relative_time=treated*relative_time
cap drop rtdum*
tabulate tr_relative_time, generate(rtdum)

forval i=2/8 {
	replace rtdum`i'=rtdum`i'-rtdum1
}

gen after=0
replace after=1 if relative_time>=9 & relative_time<=12
replace after=2 if relative_time>=13 & relative_time<=16

xtset anon relative_time

save "/homeKRTK/health_shocks/analysis_data_20250329.dta", replace

use "/homeKRTK/health_shocks/analysis_data_20250329.dta", clear
xtreg w_q i.date i.relative_time rtdum2-rtdum16 if ev_orig>=2012 & letszam~=. & FE_q~=., fe rob
lincom -rtdum2-rtdum3-rtdum4-rtdum5-rtdum6-rtdum7-rtdum8
xtreg FE_q i.date i.relative_time rtdum2-rtdum16 if ev_orig>=2012 & letszam~=. & w_q~=., fe rob
lincom -rtdum2-rtdum3-rtdum4-rtdum5-rtdum6-rtdum7-rtdum8

xtreg w_q i.date treated##ib0.after if ev_orig>=2012 & letszam~=. & FE_q~=., fe rob
xtreg FE_q i.date treated##ib0.after if ev_orig>=2012 & letszam~=. & w_q~=., fe rob

gen letszam_base_2_=letszam if relative_time==7
by anon, sort: egen letszam_base2=min(letszam_base_2)
drop letszam_base_2_

estimates clear
xtreg w_q i.date treated##ib0.after if ev_orig>=2012 & letszam~=. & FE_q~=. & (letszam_base<250 | letszam_base==.), fe rob
	est store wown_bel250
xtreg FE_q i.date treated##ib0.after if ev_orig>=2012 & letszam~=. & w_q~=. & (letszam_base<250 | letszam_base==.), fe rob
	est store FEown_bel250
xtreg w_q i.date treated##ib0.after if ev_orig>=2012 & letszam~=. & FE_q~=. & letszam_base>=250 & letszam_base!=., fe rob
	est store wown250_ab250
xtreg FE_q i.date treated##ib0.after if ev_orig>=2012 & letszam~=. & w_q~=. & letszam_base>=250 & letszam_base!=., fe rob
	est store FEown250_ab250
esttab *, cells(b(star fmt(4)) se(par(( ) fmt(4)))) legend starlevels(* 0.1 ** 0.05 *** 0.01) stats(r2 N, fmt(%9.3f %9.0g) labels(R^2))

estimates clear
xtreg w_q i.date treated##ib0.after if ev_orig>=2012 & letszam~=. & FE_q~=. & (letszam_base<123 | letszam_base==.), fe rob
	est store wown_bel250
xtreg FE_q i.date treated##ib0.after if ev_orig>=2012 & letszam~=. & w_q~=. & (letszam_base<123 | letszam_base==.), fe rob
	est store FEown_bel250
xtreg w_q i.date treated##ib0.after if ev_orig>=2012 & letszam~=. & FE_q~=. & letszam_base>=123 & letszam_base!=., fe rob
	est store wown250_ab250
xtreg FE_q i.date treated##ib0.after if ev_orig>=2012 & letszam~=. & w_q~=. & letszam_base>=123 & letszam_base!=., fe rob
	est store FEown250_ab250
esttab *, cells(b(star fmt(4)) se(par(( ) fmt(4)))) legend starlevels(* 0.1 ** 0.05 *** 0.01) stats(r2 N, fmt(%9.3f %9.0g) labels(R^2))

estimates clear
xtreg w_q i.date treated##ib0.after if ev_orig>=2012 & letszam~=. & FE_q~=. & letszam_base<619, fe rob
	est store wown_bel250
xtreg FE_q i.date treated##ib0.after if ev_orig>=2012 & letszam~=. & w_q~=. & letszam_base<619, fe rob
	est store FEown_bel250
xtreg w_q i.date treated##ib0.after if ev_orig>=2012 & letszam~=. & FE_q~=. & letszam_base>=619 & letszam_base!=., fe rob
	est store wown250_ab250
xtreg FE_q i.date treated##ib0.after if ev_orig>=2012 & letszam~=. & w_q~=. & letszam_base>=619 & letszam_base!=., fe rob
	est store FEown250_ab250
esttab *, cells(b(star fmt(5)) se(par([ ]) fmt(5)) t(par(( )) fmt(2))) legend starlevels(* 0.1 ** 0.05 *** 0.01) stats(r2 N, fmt(%9.3f %9.0g) labels(R^2))

estimates clear
xtreg w_q i.date treated##ib0.after if ev_orig>=2012 & letszam~=. & FE_q~=. & (letszam_base<500 | letszam_base==.), fe rob
	est store wown_bel250
xtreg FE_q i.date treated##ib0.after if ev_orig>=2012 & letszam~=. & w_q~=. & (letszam_base<500 | letszam_base==.), fe rob
	est store FEown_bel250
xtreg w_q i.date treated##ib0.after if ev_orig>=2012 & letszam~=. & FE_q~=. & letszam_base>=500 & letszam_base!=., fe rob
	est store wown250_ab250
xtreg FE_q i.date treated##ib0.after if ev_orig>=2012 & letszam~=. & w_q~=. & letszam_base>=500 & letszam_base!=., fe rob
	est store FEown250_ab250
esttab *, cells(b(star fmt(4)) se(par(( ) fmt(4)))) legend starlevels(* 0.1 ** 0.05 *** 0.01) stats(r2 N, fmt(%9.3f %9.0g) labels(R^2))
